from zhipuai import ZhipuAI
def test():
    text = ["这是我的测试文本"]
    client = ZhipuAI(api_key="c44665cfbed6ed17d3743f1b94469b56.eJnuhDfmJyrpVqa5")
    # 向OpenAI API请求生成嵌入向量
    response = client.embeddings.create(
        input=text,
        model='embedding-2',  # 指定使用的模型
    )

    # 从响应中提取嵌入向量并返回
    embeddings = [data.embedding for data in response.data]
    print(embeddings)

if __name__ == '__main__':
    test()